import django_filters
from django_filters.rest_framework import FilterSet

from users.models import UserInfo


class UserFilter(FilterSet):
    class Meta:
        model = UserInfo
        fields = ['id', 'username', 'qq', 'email', 'phone']


class FanFollowFilter(FilterSet):
    # TODO: fans_count, follows_count
    fans_count = django_filters.NumberFilter(label='粉丝数')
    follows_count = django_filters.NumberFilter(label='关注数')
    pass
